import os
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import JsonOutputParser

#定义提示词模板
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是一个格式转换助手，将我输入的内容转换为JSON格式。"),
    ("human", "请将以下内容转换为JSON格式：{content}")
])

model = ChatOpenAI(
    model="deepseek-chat",
    openai_api_key=os.getenv("DEEPSEEK_API_KEY"),
    openai_api_base=os.getenv("DEEPSEEK_API_BASE")
)

# 定义 JSON 输出解析器
output_parser = JsonOutputParser()

# 构建链
chain = prompt | model | output_parser

# 调用链
result = chain.invoke({
    "content": "我是晓舟，我是一个Python程序员。"
})
print(result)